-
Notifications
You must be signed in to change notification settings - Fork 89
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Split
dynamic shape parsing update
#3034
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## develop #3034 +/- ##
===========================================
+ Coverage 91.92% 91.94% +0.01%
===========================================
Files 489 489
Lines 19275 19309 +34
===========================================
+ Hits 17719 17753 +34
Misses 1556 1556 ☔ View full report in Codecov by Sentry. |
Check results before merge 🔆 |
🔴bert_large_uncased_fp16: FAILED: MIGraphX is not within tolerance - check verbose output |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Approving this PR. Some stylistic changes can be made, and if the check above is redundant (that I asked about), then please remove it.
Dont forget to fix the format issue |
auto split_axis_is_fixed = [&]() { | ||
return input_shape.dyn_dims().at(tuned_axis).is_fixed(); | ||
}; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this can simply be a bool
variable.
vec_splits.resize(num_outputs - 1, chunk_size); | ||
vec_splits.push_back(last_chunk_size); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This can potentially lead to resizing and memory allocation twice for the vector. Ideally, you can allocate vector for the splits once for the size (num_outputs) and then put values in that
Split
ONNX operator to support dynamic shapessplit
attribute or input